import os
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate
from langchain_openai import ChatOpenAI

os.environ["OPENAI_API_KEY"] = "sk-008ede2ffb134cbe93b7e2dffa5a40f9"
# 1、创建模型
model = ChatOpenAI(
    api_key="sk-008ede2ffb134cbe93b7e2dffa5a40f9",
    model='deepseek-chat',
    base_url="https://api.deepseek.com/v1"
)

# 调用示例
response = model.invoke("Python 的 lambda 函数是什么？")
print(response.content)

#
# # 2、定义提示模版
# prompt_template = ChatPromptTemplate.from_messages(
#     [("system", '请将以下的内容翻译成{language}'), ("user", "{text}")]
# )
# # 3、创建返回的数据解析器
# parser = StrOutputParser()
#
# # 4、创建链chain
# chain = prompt_template | model | parser
#
# # 5、直接使用chain进行调用
# result = chain.invoke({"language": "English", "text": '我下午还有一节课，不能去打球了'})
# print(result)

